/*******************************************************************/
/* @ All Rights Reserved Copyright (C) 2014 broada,Co.,Ltd.        */
/*                                                                 */
/*  NAME      = rtsp_comm.c                                        */
/*  NOTE      =                                                    */
/*  DATE      = 2015/07/14 by songbb                               */
/*******************************************************************/

#include <stdio.h>
#include <memory.h>
#include <sys/time.h>

#include "frmem.h"
#include "frlog.h"
#include "frreport_comm.h"
#include "frct_comm.h"
#include "frdb_cursor.h"

#include "rtsp_core.h"
#include "rtsp_comm.h"

static char hex_asc[] = "0123456789abcdef";
#define hex_asc_lo(x)   hex_asc[((x) & 0x0f)]
#define hex_asc_hi(x)   hex_asc[((x) & 0xf0) >> 4]

/*******************************************************************************
*  FUNC     :  释放内存
*  ARGS     :  pp_buffer
*  RTN      :  void
*  NOTE     :
*******************************************************************************/
void rtsp_free(void **pp_buffer)
{
    if (pp_buffer == NULL || *pp_buffer == NULL)
        return;

    frfree(*pp_buffer);
    *pp_buffer = NULL;
    return;
}

/*******************************************************************************
*  FUNC     :  时间格式化成 2015-03-29 22:25:12.448671 的格式
*  ARGS     :  *p_tv                           (时间戳)
*           :  *p_time                         (buffer)
*           :  ui_time_size                    (buffer size)
*  RTN      :
*  NOTE     :
*******************************************************************************/
FR_STATIC int __rtsp_frtime2datetimestr(struct timeval *p_tv
                        , char *p_time
                        , unsigned int ui_time_size)
{
    struct tm *p_tm = NULL;

    if(p_tv == NULL || p_time == NULL || ui_time_size <= 1)
        return FR_FAIL;

    p_tm = localtime(&(p_tv->tv_sec));

    snprintf(p_time,ui_time_size,"%04d-%02d-%02d %02d:%02d:%02d.%06ld",
             p_tm->tm_year + 1900,p_tm->tm_mon + 1,p_tm->tm_mday,
             p_tm->tm_hour,p_tm->tm_min,p_tm->tm_sec,p_tv->tv_usec);

    return FR_SUCC;
}

/*******************************************************************************
*  FUNC     :  打印时间
*  ARGS     :  st_timestamp                         (时间戳)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_print_timeval(struct timeval st_timestamp)
{
    char arr_buff[64];
    __rtsp_frtime2datetimestr(&st_timestamp, arr_buff, sizeof(arr_buff));
    printf(":[%s]\n", arr_buff);
}

/*******************************************************************************
*  FUNC     :  格式化成十六进制字符串
*  ARGS     :  *p_inbuff    (输入的字符串)
*           :  i_len        (输入的字符串长度)
*           :  *p_outbuff   (格式化后的十六进制字符串)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_bin_to_hex_string(char *p_inbuff
                            , int i_len
                            , char *p_outbuff)
{
    int i_loop;
    char c;
    int i_used = 0;

    for (i_loop = 0; i_loop < i_len; i_loop++)
    {
        c = p_inbuff[i_loop];
        p_outbuff[i_used++] = hex_asc_hi(c);
        p_outbuff[i_used++] = hex_asc_lo(c);
    }

    return;
}

/*******************************************************************************
*  FUNC     :  按一个字节打印十六进行字符串
*  ARGS     :  *p_buff  (char*字符串)
*           :  i_len   (字符串长度)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_print_hex_string(char *p_buff
                        , int i_len)
{
    int i_loop;

    for (i_loop = 0; i_loop < i_len; i_loop++)
        printf("%02x", p_buff[i_loop] & 0xFF);

    return;
}

/*******************************************************************************
*  FUNC     :  打印指定位置的包内容
*  ARGS     :  *p_st_rtsp_packet                     (RTSP数据包)
*           :  i_pos                                (开始位置)
*           :  i_len                                (长度)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_print_packet_specified(L7_rtsp_packet *p_st_rtsp_packet
                                , int i_pos
                                , int i_len)
{
    if (p_st_rtsp_packet == NULL)
        return;
    char *p_buff_tmp = NULL;
    if (bytebuff_show(p_st_rtsp_packet->p_st_packet, i_pos, i_len, &p_buff_tmp) != FR_SUCC)
        return;
    rtsp_print_hex_string(p_buff_tmp, i_len);
    rtsp_free((void **) &p_buff_tmp);
    return;
}

/*******************************************************************************
*  FUNC     :  打印当前位置开始的包内容
*  ARGS     :  *p_st_rtsp_packet                     (RTSP数据包)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_print_packet_current(L7_rtsp_packet *p_st_rtsp_packet)
{
    if (p_st_rtsp_packet == NULL)
        return;
    char *p_buff_tmp = NULL;
    if (bytebuff_show(p_st_rtsp_packet->p_st_packet, bytebuff_getPosition(p_st_rtsp_packet->p_st_packet), bytebuff_remaining(p_st_rtsp_packet->p_st_packet), &p_buff_tmp) != FR_SUCC)
        return;
    rtsp_print_hex_string(p_buff_tmp, bytebuff_remaining(p_st_rtsp_packet->p_st_packet));
    rtsp_free((void **) &p_buff_tmp);
    return;
}

/*******************************************************************************
*  FUNC     :  打印整个包内容
*  ARGS     :  *p_st_rtsp_packet                     (RTSP数据包)
*  RTN      :
*  NOTE     :
*******************************************************************************/
void rtsp_print_packet(L7_rtsp_packet *p_st_rtsp_packet)
{
    if (p_st_rtsp_packet == NULL)
        return;
    char *p_buff_tmp = NULL;
    if (bytebuff_show(p_st_rtsp_packet->p_st_packet, 0, bytebuff_length(p_st_rtsp_packet->p_st_packet), &p_buff_tmp) != FR_SUCC)
        return;
    rtsp_print_hex_string(p_buff_tmp, bytebuff_length(p_st_rtsp_packet->p_st_packet));
    rtsp_free((void **) &p_buff_tmp);
    return;
}

/*******************************************************************************
*  FUNC     :  判断是否处于分片包状态
*  ARGS     :  *p_st_rtsp_packet                        (RTSP数据包)
*  RTN      :  0                                        (FR_NO)
*              1                                        (FR_YES)
*  NOTE     :
*******************************************************************************/
int rtsp_packet_is_in_fragment(L7_rtsp_packet *p_st_rtsp_packet)
{
    if (p_st_rtsp_packet == NULL)
        return FR_NO;

    return (p_st_rtsp_packet->fragment_state == APM_RTSP_FRAGMENT_NEED_NEXT_REQ_PACKET
            || p_st_rtsp_packet->fragment_state == APM_RTSP_FRAGMENT_NEED_NEXT_RESP_PACKET) ? FR_YES : FR_NO;
}

/*******************************************************************************
*  FUNC     :  上报通信消息的响应信息
*  ARGS     :  *p_st_conn                               (ct信息)
*           :  *p_st_hash_report                        (数据上报)
*  RTN      :  0                                        (FR_SUCC)
*              1                                        (FR_FAIL)
*  NOTE     :
*******************************************************************************/
int rtsp_report_data(Conntrack *p_st_conn
                    , int i_is_req
                    , Frreport_body_data *p_st_body
                    , Frhash_header *p_st_hash_report)
{
    RTSP_PRINT_INFO("rtsp_report_data\n");
    Frreport_data *p_st_report_data = NULL;

    if ((p_st_report_data = frreport_data_l7_new(i_is_req == FR_YES ? FRREPORT_TYPE_L7DATA_REQ : FRREPORT_TYPE_L7DATA_RESP)) == NULL)
        goto error;

    if (frreport_data_add_body(p_st_report_data, p_st_body) != FR_SUCC)
        goto error;

    if (frreport_data_hash(p_st_hash_report, p_st_report_data) != FR_SUCC)
        goto error;

    return FR_SUCC;
error:
    frreport_data_free(&p_st_report_data);
    return FR_FAIL;
}

/*******************************************************************************
*  FUNC     :  释放 L7_rtsp_stmt
*  ARGS     :  **pp_st_rtsp_stmt                             (L7_rtsp_stmt)
*  RTN      :  void
*  NOTE     :
*******************************************************************************/
void rtsp_free_stmt(L7_rtsp_stmt **pp_st_rtsp_stmt)
{
    if (pp_st_rtsp_stmt == NULL || *pp_st_rtsp_stmt == NULL)
        return;
    rtsp_free((void **) pp_st_rtsp_stmt);

    return;
}

/*******************************************************************************
*  FUNC     :  初始化 L7_rtsp_stmt
*  ARGS     :  *p_st_rtsp_stmt                               (L7_rtsp_stmt)
*  RTN      :  void
*  NOTE     :
*******************************************************************************/
void rtsp_init_stmt(L7_rtsp_stmt *p_st_rtsp_stmt)
{
    if (p_st_rtsp_stmt == NULL)
        return;

    p_st_rtsp_stmt->em_req_method = RTSP_METHOD_TYPE_UNKNOWN;
    memset(&p_st_rtsp_stmt->st_req_end_timestamp, 0x0, sizeof(struct timeval));
    memset(&p_st_rtsp_stmt->st_resp_start_timestamp, 0x0, sizeof(struct timeval));
    p_st_rtsp_stmt->ul_resp_server_time_us = 0L;
    p_st_rtsp_stmt->ul_resp_bytes = 0L;

    return;
}
